from peewee import *
from datetime import datetime
import os


'''
Sheet：表格的属性包括
    name - 对应企业的名字
    catagory - ‘资产负债表’、‘利润表’、‘现金流量表’、‘财务指标表’
Item：科目
    name - 科目名字，比如‘总资产’
    sheet - 外键，属于那个表，后面利于汇总计算
    belong - 科目所属的，比如应收账款属于总资产
    year3 - 前3年的年度数据
    year2 - 前2年的年度数据
    year1 - 前1年的年度数据
    month - 最近1期的月度数据
    delta - 差值
    ratio - 变化率
    summa - 和值
    avera - 平均值
'''

def database(db_path):
    db = SqliteDatabase(db_path)
    return db

db = ''

class BaseModel(Model):
    '''
    使用之前需要定义全局变量：db
    '''
    class Meta:
        Datebase = db


class Sheet(BaseModel):
    name = CharField()
    form = CharField()
    create_date = DateTimeField(default = datetime.now())
    

class Item(BaseModel):
    name = CharField()
    sheet = ForeignKeyField(Sheet, backref='items')
    belong = CharField()
    year3 = FloatField()
    year2 = FloatField()
    year1 = FloatField()
    month = FloatField()
    delta = FloatField()
    ratio = FloatField()
    sigma = FloatField()
    avera = FloatField()


